from flask import g

'''
自定义一个装饰器，验证某些请求是否已经登陆过了。如果已经登陆了继续访问。
本质上就是一个登陆拦截
'''


def login_required(func):
    def wrapper(*args, **kwargs):
        if g.user_id is not None:  # 如果是登陆过的，在获取token的请求钩子里面就已经给g.uer_id 赋值了。
            return func(*args, **kwargs)
        else:
            return {'message': 'Invalid Token.'}, 401

    return wrapper
